CUDA Toolkit on Docker

Pre-requisites

• Docker
• Nvidia proprietary driver
• Nvidia container toolkit

On NixOS, adding the Nvidia container toolkit is simple

virtualisation.docker = {
  enable = true;
  enableNvidia = enableNvidia;
};

Getting Container

• docker run -it --rm -v $(realpath ~/notebooks):/tf/notebooks -p 8888:8888 tensorflow/tensorflow:latest-gpu-jupyter